cd ../gt
nohup python train.py --data_type AFHQDataset \
    --data_args "{'power_of_two': True}" \
    --results_dir ../results/AFHQ/gt --input_shape 3 64 64 \
    --post_model_type CondConvINN2 --post_model_args "{'num_conv_layers':[2, 4, 4], 'num_fc_layers':[4], 'cond_layer_thicknesses':[32, 64, 64, 256]}" \
    --reg_parameter 1 --lr_post 1e-04 --batch_size 1024 \
    --num_bits 0 --num_iters_post 200000 --gpu 2 --num_z 1 \
    --degradation_type GaussianNoise \
    --degradation_args "{'mean':0., 'std':0.1}" \
    --reg_tv 0 \
    --num_iters 5000 > /home/baiweimin/yifei/flow-diff/log/flow/gt_train_denoise_nocon.log 2>&1 &

    # --degradation_type GaussianBlurNoise \
    # --degradation_args "{'kernel_sigma':1, 'mean':0., 'std':0.01}" \